package com.gitee.helioz.openai.api.dto.embeddings;

import com.gitee.helioz.openai.api.dto.UsageDTO;
import lombok.Data;

import java.util.List;

/**
 * 〈嵌入响应〉
 *
 * @author <a href="https://gitee.com/helioz">Morning Starr</a>
 */
@Data
public class EmbeddingResponse {

    /**返回结果的对象类型，固定为 "list"。*/
    private String object;
    /**包含嵌套的 embedding 对象的数组。*/
    private List<Embedding> data;
    /**使用的模型的标识符。*/
    private String model;
    /**包含有关 API 调用中使用的令牌数的信息。*/
    private UsageDTO usage;

    @Data
    public static class Embedding {
        /**嵌套对象的类型，固定为 "embedding"。*/
        private String object;
        /**表示输入文本的嵌入向量的浮点数数组。*/
        private List<Float> embedding;
        /**输入文本在请求中的索引。*/
        private Integer index;
    }
}
